Skip to content

[AKS] az aks nodepool update: Add GPU driver install options install and none for --gpu-driver parameter#9472

Merged
yanzhudd merged 13 commits intoAzure:mainfrom
lilypan26:lily/update-gpu-driver
Dec 19, 2025
Merged

[AKS] az aks nodepool update: Add GPU driver install options install and none for --gpu-driver parameter#9472
yanzhudd merged 13 commits intoAzure:mainfrom
lilypan26:lily/update-gpu-driver

Conversation

@lilypan26
Copy link
Copy Markdown
Contributor


This checklist is used to make sure that common guidelines for a pull request are followed.

Related command

az aks update --gpu-driver none

General Guidelines

  • Have you run azdev style <YOUR_EXT> locally? (pip install azdev required)
  • Have you run python scripts/ci/test_index.py -q locally? (pip install wheel==0.30.0 required)
  • My extension version conforms to the Extension version schema

For new extensions:

About Extension Publish

There is a pipeline to automatically build, upload and publish extension wheels.
Once your pull request is merged into main branch, a new pull request will be created to update src/index.json automatically.
You only need to update the version information in file setup.py and historical information in file HISTORY.rst in your PR but do not modify src/index.json.

@azure-client-tools-bot-prd
Copy link
Copy Markdown

azure-client-tools-bot-prd bot commented Dec 12, 2025

⚠️Azure CLI Extensions Breaking Change Test
⚠️aks-preview
rule cmd_name rule_message suggest_message
⚠️ 1006 - ParaAdd aks nodepool update cmd aks nodepool update added parameter gpu_driver

@yonzhan
Copy link
Copy Markdown
Collaborator

yonzhan commented Dec 12, 2025

Thank you for your contribution! We will review the pull request and get back to you soon.

@azure-client-tools-bot-prd
Copy link
Copy Markdown

Hi @lilypan26,
Please write the description of changes which can be perceived by customers into HISTORY.rst.
If you want to release a new extension version, please update the version in setup.py as well.

@github-actions
Copy link
Copy Markdown
Contributor

The git hooks are available for azure-cli and azure-cli-extensions repos. They could help you run required checks before creating the PR.

Please sync the latest code with latest dev branch (for azure-cli) or main branch (for azure-cli-extensions).
After that please run the following commands to enable git hooks:

pip install azdev --upgrade
azdev setup -c <your azure-cli repo path> -r <your azure-cli-extensions repo path>

@github-actions
Copy link
Copy Markdown
Contributor

CodeGen Tools Feedback Collection

Thank you for using our CodeGen tool. We value your feedback, and we would like to know how we can improve our product. Please take a few minutes to fill our codegen survey

@github-actions
Copy link
Copy Markdown
Contributor

github-actions bot commented Dec 12, 2025

@lilypan26 lilypan26 marked this pull request as ready for review December 12, 2025 00:55
Copilot AI review requested due to automatic review settings December 12, 2025 00:55
Copy link
Copy Markdown
Contributor

Copilot AI left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Pull request overview

This PR adds the --gpu-driver parameter to the az aks nodepool update command to control GPU driver installation on node pools. The parameter accepts values "Install" or "None" to either install or skip GPU driver installation.

Key Changes:

  • Added gpu_driver parameter configuration to az aks nodepool update command in _params.py
  • Updated HISTORY.rst to document the new feature in version 19.0.0b19

Reviewed changes

Copilot reviewed 2 out of 2 changed files in this pull request and generated 2 comments.

File Description
src/aks-preview/azext_aks_preview/_params.py Adds gpu_driver parameter configuration to the aks nodepool update argument context using the existing gpu_driver_install_modes enum
src/aks-preview/HISTORY.rst Documents the addition of --gpu-driver option to az aks nodepool update command in version 19.0.0b19

)
c.argument(
"gpu_driver",
arg_type=get_enum_type(gpu_driver_install_modes)
Copy link

Copilot AI Dec 12, 2025

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

The gpu_driver parameter is missing documentation. Add a help parameter to provide guidance to users, similar to line 1978. Based on the help text in _help.py line 2162, consider adding: help="Whether to install driver for GPU node pool. Possible values are 'Install' or 'None'. Default is 'Install'."

Suggested change
arg_type=get_enum_type(gpu_driver_install_modes)
arg_type=get_enum_type(gpu_driver_install_modes),
help="Whether to install driver for GPU node pool. Possible values are 'Install' or 'None'. Default is 'Install'."

Copilot uses AI. Check for mistakes.
@FumingZhang
Copy link
Copy Markdown
Member

please fix failed CI checks GH Azdev Linter / azdev-linter (pull_request_target)

  • FAIL - HIGH severity: missing_parameter_help
    Parameter: aks nodepool update, gpu_driver - Missing help

Copy link
Copy Markdown
Member

@FumingZhang FumingZhang left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

As mentioned in this comment, you also need to update the backend AKSAgentPoolUpdateDecorator

@lilypan26 lilypan26 force-pushed the lily/update-gpu-driver branch from 5740655 to 7973fe2 Compare December 17, 2025 23:12
@github-actions github-actions bot added the release-version-block Updates do not qualify release version rules. NOTE: please do not edit it manually. label Dec 18, 2025
@github-actions github-actions bot removed the release-version-block Updates do not qualify release version rules. NOTE: please do not edit it manually. label Dec 18, 2025
@yanzhudd yanzhudd merged commit 246bd0c into Azure:main Dec 19, 2025
24 checks passed
@azclibot
Copy link
Copy Markdown
Collaborator

[Release] Update index.json for extension [ aks-preview-19.0.0b18 ] : https://dev.azure.com/msazure/One/_build/results?buildId=147195849&view=results

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

AKS Auto-Assign Auto assign by bot

Projects

None yet

Development

Successfully merging this pull request may close these issues.

7 participants